from collections import OrderedDict

from goods.models import GoodsChannel


# 封装商品分类信息
def get_categories():
    # 准备商品对应字典
    categories = OrderedDict()

    # 查询商品的所有频道
    channels = GoodsChannel.objects.order_by('group_id', 'sequence')

    # 遍历所有频道频道
    for channel in channels:
        # 获取当前频道所在组
        group_id = channel.group_id
        # 判断group_id是否以存在在categories中，不存在的则添加
        if group_id not in categories:
            # 构造基本的数据框架
            categories[group_id] = {'channels': [], 'sub_cats': []}

        # 查询当前频道对象的一级类别
        cat1 = channel.category

        # 追加当前频道
        categories[group_id]['channels'].append({
            'id': cat1.id,
            'name': cat1.name,
            'url': channel.url
        })
        # 构建当前类别的子类别2级3级
        for cat2 in cat1.subs.all():
            cat2.sub_cats = []  # 追加二级数据所需要的列表
            for cat3 in cat2.subs.all():  # 从二级找3级
                cat2.sub_cats.append(cat3)  # 三级添加到2级sub_cats


            categories[group_id]['sub_cats'].append(cat2)

    return categories
